347. Top K Frequent Elements - LeetCode Solution

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        arr = []
        dict1 = {}
        
        for num in nums:
            if num in dict1:
                dict1[num] +=1
                
            else:
                dict1[num] = 1
        for i in dict1:
            arr.append([dict1[i], i])
        
        arr.sort(reverse  = True)
        ans = []
        for i in range(k):
            ans.append(arr[i][1])
        return ans